关于CUDA中的warp shuffle函数说明

您所在的位置:网站首页 GPU warp什么意思 关于CUDA中的warp shuffle函数说明

关于CUDA中的warp shuffle函数说明

2024-07-02 21:59| 来源: 网络整理| 查看: 265

以前听过山本前辈【山本A梦我对不起你Orz】细心的讲解。然后忘了。

现在重新理解了一遍、把自己的理解中的要点记了下来

基本的内容(原文):CUDA C PROGRAMMING GUIDE - warp shuffle 函数: https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#warp-shuffle-functions

 

 

在使用warp shuffle函数时,比如实际用的时候用的是__shfl_up_sync(0xffffffff, res, i),如下面一段计算前缀和(prefix sum)的代码:

1 #include 2 #include 3 #include "cuda_runtime_api.h" 4 #define MASK 0xffffffff 5 6 __global__ void doKernel(int* A, int* B){ 7 ¦ int tid = threadIdx.x; 8 //¦ int val = A[tid]; 9 ¦ int val = 1; 10 ¦ int res = val; 11 ¦ int tmp; 12 ¦ /* 13 ¦ for (int k = 1; k


【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3